LogicalAnd

逐元素执行逻辑与（Logical AND）运算，支持广播。

对输入张量的对应元素先进行布尔化判断，再执行逻辑与运算，输出结果为 0 或 1。

\[\begin{split}\text{output}_i = \begin{cases} 1, & \text{if } (\text{input0}_i \neq 0) \land (\text{input1}_i \neq 0) \\ 0, & \text{otherwise} \end{cases}\end{split}\]
输入：

  • input0 - 第一个输入张量的数据地址。

  • input1 - 第二个输入张量的数据地址。

  • output - 输出张量的数据地址。

  • params - 参数打包成数组（7个元素），格式如下：

    • params[0]: input0_dims 地址

    • params[1]: input1_dims 地址

    • params[2]: output_dims 地址

    • params[3]: strides0 地址（需分配 num_dims*sizeof(int) 空间）

    • params[4]: strides1 地址（需分配 num_dims*sizeof(int) 空间）

    • params[5]: strides_output 地址（需分配 num_dims*sizeof(int) 空间）

    • params[6]: num_dims（张量维度数）

  • core_mask - 核掩码（仅共享存储版本）。

输出：

  • output - 输出结果地址，0 或 1。

支持平台：

FT78NE MT7004

备注

  • FT78NE 支持 fp32, fp64, int8, int16, int32

  • MT7004 支持 fp16, fp32, int16, int32

  • 输入值在参与逻辑运算前会被转换为布尔值（非 0 为 true，0 为 false）

  • 输出结果仅为 0 或 1

共享存储版本：

void fp_logical_and_s(float *input0, float *input1, float *output, unsigned long long *params, int core_mask)
void hp_logical_and_s(half *input0, half *input1, half *output, unsigned long long *params, int core_mask)
void dp_logical_and_s(double *input0, double *input1, double *output, unsigned long long *params, int core_mask)
void i8_logical_and_s(int8_t *input0, int8_t *input1, int8_t *output, unsigned long long *params, int core_mask)
void i16_logical_and_s(int16_t *input0, int16_t *input1, int16_t *output, unsigned long long *params, int core_mask)
void i32_logical_and_s(int32_t *input0, int32_t *input1, int32_t *output, unsigned long long *params, int core_mask)

C调用示例：

 1// FT78NE 多核示例
 2#include <stdio.h>
 3
 4int main(int argc, char* argv[]) {
 5    float *input0 = (float *)0x81000000;
 6    float *input1 = (float *)0x82000000;
 7    float *output = (float *)0x83000000;
 8    int *strides0 = (int *)0x84000000;
 9    int *strides1 = (int *)0x85000000;
10    int *strides_output = (int *)0x86000000;
11
12    int input0_dims[] = {4, 1, 16};
13    int input1_dims[] = {4, 8, 16};
14    int output_dims[] = {4, 8, 16};
15    int num_dims = 3;
16    int core_mask = 0xff;
17
18    unsigned long long params[7];
19    params[0] = (unsigned long long)input0_dims;
20    params[1] = (unsigned long long)input1_dims;
21    params[2] = (unsigned long long)output_dims;
22    params[3] = (unsigned long long)strides0;
23    params[4] = (unsigned long long)strides1;
24    params[5] = (unsigned long long)strides_output;
25    params[6] = (unsigned long long)num_dims;
26
27    fp_logical_and_s(input0, input1, output, params, core_mask);
28    return 0;
29}

私有存储版本：

void fp_logical_and_p(float *input0, float *input1, float *output, unsigned long long *params)
void hp_logical_and_p(half *input0, half *input1, half *output, unsigned long long *params)
void dp_logical_and_p(double *input0, double *input1, double *output, unsigned long long *params)
void i8_logical_and_p(int8_t *input0, int8_t *input1, int8_t *output, unsigned long long *params)
void i16_logical_and_p(int16_t *input0, int16_t *input1, int16_t *output, unsigned long long *params)
void i32_logical_and_p(int32_t *input0, int32_t *input1, int32_t *output, unsigned long long *params)

C调用示例：

 1// FT04 单核示例
 2#include <stdio.h>
 3
 4int main(int argc, char* argv[]) {
 5    float *input0 = (float *)0x10010000;
 6    float *input1 = (float *)0x10020000;
 7    float *output = (float *)0x10030000;
 8    int *strides0 = (int *)0x10050000;
 9    int *strides1 = (int *)0x10053000;
10    int *strides_output = (int *)0x10056000;
11
12    int input0_dims[] = {4, 1, 16};
13    int input1_dims[] = {4, 8, 16};
14    int output_dims[] = {4, 8, 16};
15    int num_dims = 3;
16
17    unsigned long long params[7];
18    params[0] = (unsigned long long)input0_dims;
19    params[1] = (unsigned long long)input1_dims;
20    params[2] = (unsigned long long)output_dims;
21    params[3] = (unsigned long long)strides0;
22    params[4] = (unsigned long long)strides1;
23    params[5] = (unsigned long long)strides_output;
24    params[6] = (unsigned long long)num_dims;
25
26    fp_logical_and_p(input0, input1, output, params);
27    return 0;
28}